/**
 * Software License, Version 1.0
 * 
 * Copyright 2003 The Trustees of Indiana University.  All rights reserved.
 * 
 *
 *Redistribution and use in source and binary forms, with or without 
 *modification, are permitted provided that the following conditions are met:
 *
 *1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license;
 *2) All redistributions in binary form must reproduce the above copyright 
 * notice, this list of conditions and the disclaimer listed in this license in
 * the documentation and/or other materials provided with the distribution;
 *3) Any documentation included with all redistributions must include the 
 * following acknowledgement:
 *
 *"This product includes software developed by the Community Grids Lab. For 
 * further information contact the Community Grids Lab at 
 * http://communitygrids.iu.edu/."
 *
 * Alternatively, this acknowledgement may appear in the software itself, and 
 * wherever such third-party acknowledgments normally appear.
 * 
 *4) The name Indiana University or Community Grids Lab or NaradaBrokering, 
 * shall not be used to endorse or promote products derived from this software 
 * without prior written permission from Indiana University.  For written 
 * permission, please contact the Advanced Research and Technology Institute 
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202.
 *5) Products derived from this software may not be called NaradaBrokering, 
 * nor may Indiana University or Community Grids Lab or NaradaBrokering appear
 * in their name, without prior written permission of ARTI.
 * 
 *
 * Indiana University provides no reassurances that the source code provided 
 * does not infringe the patent or any other intellectual property rights of 
 * any other entity.  Indiana University disclaims any liability to any 
 * recipient for claims brought by any other entity based on infringement of 
 * intellectual property rights or otherwise.  
 *
 *LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO 
 *WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 *NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF 
 *INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS. 
 *INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS", 
 *"VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.  
 *LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR 
 *ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION 
 *GENERATED USING SOFTWARE.
 */
package cgl.narada.wsinfra.deployment.axis.wsrm;

import java.io.ByteArrayInputStream;
import java.io.InputStream;

import javax.xml.namespace.QName;

import org.apache.axis.message.SOAPEnvelope;
import org.apache.xmlbeans.XmlCursor;

import cgl.narada.wsinfra.schemas.addressing.ActionDocument;
import cgl.narada.wsinfra.schemas.addressing.EndpointReferenceType;
import cgl.narada.wsinfra.schemas.addressing.FromDocument;
import cgl.narada.wsinfra.schemas.addressing.RelatesToDocument;
import cgl.narada.wsinfra.schemas.soap.EnvelopeDocument;
import cgl.narada.wsinfra.util.SoapEnvelopeConversion;
import cgl.narada.wsinfra.util.SoapMessageAlteration;
import cgl.narada.wsinfra.wsa.WsaEnvelopeCreator;
import cgl.narada.wsinfra.wsa.WsaEprCreator;
import cgl.narada.wsinfra.wsa.WsaProcessingFactory;
import cgl.narada.wsinfra.wsrm.WsrmQNames;

/**
 *Creates an SOAPEnvelope. Both handler and Proxy base web service deployments
 * are using message type web service. Message type web services work with
 * SOAPEnvelope. This class helps to create them.
 *
 * @author Beytullah Yildiz
 * $Date: 2005/07/29 22:45:40 $
 * $Revision: 1.1 $
 */

public class WsrmCreateEnvelope {
  private String moduleName = "WsrmCreateEnvelopeDocument";

  private WsaEprCreator wsaEprCreator;
  private WsaEnvelopeCreator wsaEnvelopeCreator;
  private SoapEnvelopeConversion soapEnvelopeConversion;
  private SoapMessageAlteration soapMessageAlteration;
  private int tracker;

  public WsrmCreateEnvelope() {

    wsaEprCreator = WsaProcessingFactory.getWsaEprCreator();
    wsaEnvelopeCreator = WsaProcessingFactory.getWsaEnvelopeCreator();
    soapEnvelopeConversion = SoapEnvelopeConversion.getInstance();

  }

  /**
   * Creates EnvelopeDocument by using source and sink adresses
   */
  public EnvelopeDocument createEnvDocument(String source, String sink) {

    EndpointReferenceType sourceEpr = wsaEprCreator.createEpr(source);
    EndpointReferenceType sinkEpr = wsaEprCreator.createEpr(sink);

    FromDocument from = FromDocument.Factory.newInstance();
    from.setFrom(sourceEpr);
    ActionDocument action = null;
    RelatesToDocument relatesTo = null;

    EnvelopeDocument envelopeDocument =
        wsaEnvelopeCreator.createSoapEnvelope(sinkEpr, from, action,
                                              relatesTo);

    return envelopeDocument;

  }

  /**
   * Adds Qname into both header and body. We may remove adding any QName into
   * body. Howover, axis does not send any message which has empty body
   */
  public EnvelopeDocument addQName(EnvelopeDocument envelopeDocument,
                                   String qString) {
    boolean lastMessage = false;
    QName qname = new QName("http://www.naradabrokering.org", qString);

    XmlCursor headerCursor =
        envelopeDocument.getEnvelope().getHeader().newCursor();

    String value = "Tracker :" + Math.random();
    headerCursor.toNextToken();
    headerCursor.insertElementWithText(qname, value);
    headerCursor.dispose();

    //adds element to the body.
    XmlCursor bodyCursor =
        envelopeDocument.getEnvelope().getBody().newCursor();
    bodyCursor.toNextToken();
    bodyCursor.insertElementWithText(qname, value);
    bodyCursor.dispose();

    if (lastMessage) {
      QName qNameNbx = WsrmQNames.getInstance().getLastMessageNBX();
      String valueNbx = "true";
      boolean added = soapMessageAlteration.addToSoapHeader(envelopeDocument,
          qNameNbx, valueNbx);
      if (added) {
        System.out.println(moduleName + "LAST MESSAGE SUCCESS");
      }
    }

    return envelopeDocument;

  }

  /**
   * returns SOAPEnvelope
   */
  public SOAPEnvelope getEnvelope(String messageType, String source,
                                  String sink) {

    EnvelopeDocument envelopeDocument = createEnvDocument(source, sink);
    envelopeDocument = addQName(envelopeDocument, messageType);
    String soapString = envelopeDocument.xmlText();

    InputStream inputStream = new ByteArrayInputStream(soapString.getBytes());
    SOAPEnvelope soapEnvelope = null;
    try {
      soapEnvelope = new SOAPEnvelope(inputStream);
    }
    catch (Exception e) {
      System.out.println(e);
    }
    return soapEnvelope;

  }

}